package com.zh.note.leetcode.dp;

/**
 * 123. 买卖股票的最佳时机 III
 * dp[i][0] = Math.max(dp[i - 1][0], dp[i-1][1]-prices[i]);
 * dp[i][1] = Math.max(dp[i - 1][1], dp[i - 1][0] + prices[i]);
 */
public class LC_123_maxProfit {

    public int maxProfit(int[] prices) {
        if (prices == null || prices.length == 0) {
            return 0;
        }
        //dp[i][0]，下标为 i 这一天结束的时候，不操作，我们持有的现金数
        //dp[i][1] 下标为 i 这一天结束的时候，第一次持有股票时，我们持有的现金数
        //dp[i][2] 下标为 i 这一天结束的时候，第一次不持有股票时，我们持有的现金数
        //dp[i][3] 下标为 i 这一天结束的时候，第二次持有股票时，我们持有的现金数
        //dp[i][4] 下标为 i 这一天结束的时候，第二次不持有股票时，我们持有的现金数
        int[][] dp = new int[prices.length][5];
        dp[0][1] = -prices[0];
        // 同一天买卖
        dp[0][2] = 0;
        // 初始化第二次买入的状态是确保 最后结果是最多两次买卖的最大利润
        dp[0][3] = -prices[0];
        for (int i = 1; i < prices.length; i++) {
            dp[i][1] = Math.max(dp[i - 1][1], -prices[i]);
            dp[i][2] = Math.max(dp[i - 1][2], dp[i - 1][1] + prices[i]);
            dp[i][3] = Math.max(dp[i - 1][3], dp[i - 1][2] - prices[i]);
            dp[i][4] = Math.max(dp[i - 1][4], dp[i - 1][3] + prices[i]);
        }
        return dp[prices.length - 1][4];
    }

    public int maxProfit2(int[] prices) {
        if (prices == null || prices.length == 0) {
            return 0;
        }
        int[] dp = new int[5];
        dp[1] = -prices[0];
        dp[3] = -prices[0];
        for (int i = 1; i < prices.length; i++) {
            dp[1] = Math.max(dp[1], dp[0] - prices[i]);
            dp[2] = Math.max(dp[2], dp[1] + prices[i]);
            dp[3] = Math.max(dp[3], dp[2] - prices[i]);
            dp[4] = Math.max(dp[4], dp[3] + prices[i]);
        }
        return dp[4];
    }
}
